# ===============================================================#
#                          平滑速度衰减函数
# ===============================================================#
import math


def drap(r, a, p):
    if r < 0:
        return 0
    else:
        if r * p > 0 and r * p < a / p:
            return r * p
        else:
            return math.sqrt(2*a*r - (a/p)**2)

